#include "SQLAlterTableAddColumn.h"
#include "../../visitor/SQLASTVisitor.h"
#include "../../../utils/listTran.h"
#include "SQLColumnDefinition.h"

SQLAlterTableAddColumn::SQLAlterTableAddColumn()
{
  columns = std::make_shared<std::list<SQLColumnDefinition_ptr>>();
}

// @Override
void SQLAlterTableAddColumn::accept0(SQLASTVisitor_ptr visitor)
{
  if (visitor->visit(SharedObject(SQLAlterTableAddColumn)))
  {
    SQLObject_list_ptr tmp = ListTran::tran_class<SQLColumnDefinition, SQLObject>(columns);
    acceptChild(visitor, tmp);
  }
  visitor->endVisit(SharedObject(SQLAlterTableAddColumn));
}

SQLColumnDefinition_list_ptr  SQLAlterTableAddColumn::getColumns()
{
  return columns;
}

void SQLAlterTableAddColumn::addColumn(SQLColumnDefinition_ptr column)
{
  if (column != nullptr)
  {
    column->setParent(SharedObject(SQLObjectImpl));
  }
  this->columns->push_back(column);
}

SQLName_ptr SQLAlterTableAddColumn::getFirstColumn()
{
  return firstColumn;
}

void SQLAlterTableAddColumn::setFirstColumn(SQLName_ptr first)
{
  this->firstColumn = first;
}

BOOL_ptr SQLAlterTableAddColumn::isFirst()
{
  return first;
}

void SQLAlterTableAddColumn::setFirst(BOOL_ptr first)
{
  this->first = first;
}

SQLName_ptr SQLAlterTableAddColumn::getAfterColumn()
{
  return afterColumn;
}

void SQLAlterTableAddColumn::setAfterColumn(SQLName_ptr after)
{
  this->afterColumn = after;
}

BOOL_ptr SQLAlterTableAddColumn::getRestrict()
{
  return restrict;
}

BOOL_ptr SQLAlterTableAddColumn::isRestrict()
{
  // if (restrict == null) {
  //     return !cascade;
  // }
  return restrict;
}

void SQLAlterTableAddColumn::setRestrict(BOOL_ptr restrict)
{
  this->restrict = restrict;
}

BOOL_ptr SQLAlterTableAddColumn::isCascade()
{
  return cascade;
}

void SQLAlterTableAddColumn::setCascade(BOOL_ptr cascade)
{
  this->cascade = cascade;
}
